package io.bdmc.common.utils;

import org.apache.http.HttpStatus;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

/**
 * 返回数据
 * 
 * @author huang lu
 * @email hlkyo@qq.com
 * @date 2016年10月27日 下午9:59:27
 */
public class R extends HashMap<String, Object> {
	private static final long serialVersionUID = 1L;

	public R() {
		put("code", 0);
	}

	public static R error() {
		return error(HttpStatus.SC_INTERNAL_SERVER_ERROR, "未知异常，请联系管理员");
	}

	public static R error(String msg) {
		return error(HttpStatus.SC_INTERNAL_SERVER_ERROR, msg);
	}

	public static R error(int code, String msg) {
		R r = new R();
		r.put("code", code);
		r.put("msg", msg);
		return r;
	}

	public static R ok(String msg) {
		R r = new R();
		r.put("msg", msg);
		return r;
	}

	public static R ok(Map<String, Object> map) {
		R r = new R();
		r.putAll(map);
		return r;
	}

	public static R ok() {
		return new R();
	}

	public static R trueOrFalse(Boolean b) {
		if (b) {
			return ok();
		} else {
			return error(-1, "操作失败");
		}

	}

	public R put(String key, Object value) {
		super.put(key, value);
		return this;
	}

	public R putData(Object value) {
		super.put("data", value);
		return this;
	}

	public R putPageData(Collection<?> pageData, long total) {
		HashMap<String, Object> map = new HashMap<>(2);
		map.put("total", total);
		map.put("records", pageData);
		super.put("data", map);
		return this;
	}
}
